package cn.jeefast.common.utils;

import com.baomidou.mybatisplus.plugins.Page;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 返回数据
 *
 * @author theodo
 * @email 36780272@qq.com
 * @date 2017年10月27日 下午9:59:27
 */
public class R extends HashMap<String, Object> {
	private static final long serialVersionUID = 1L;

	public R() {
		put("code", 0);
	}

	public static R error() {
		return error(500, "未知异常，请联系管理员");
	}

	public static R error(String message) {
		return error(500, message);
	}

	public static R error(int code, String message) {
		R r = new R();
		r.put("code", code);
		r.put("message", message);
		return r;
	}

	public static R ok(String message) {
		R r = new R();
		r.put("message", message);
		return r;
	}

	public static R ok(Map<String, Object> map) {
		R r = new R();
		r.putAll(map);
		return r;
	}

	public static R ok() {
		return new R();
	}

	public R put(String key, Object value) {
		super.put(key, value);
		return this;
	}

	public R putAll(Page page) {
		List list = page.getRecords();
    long count = page.getTotal();
		Map<String, Object> condition = page.getCondition();
		int current = page.getCurrent();
		int limit = page.getLimit();
    long pages = page.getPages();
		int size = page.getSize();
		return R.ok().put("data", list).
				put("count", count).
				put("condition", condition).
				put("current", current).
				put("limit", limit).
				put("pages", pages).
				put("size", size);
	}
}
